package homework1_array.alg_33;

import static homework1_array.Utils.printList;

public class ArrayHomework {

    public static void main(String[] args) {

        //todo 作业1：完成这个测试方法中的addByElementBySequence()方法
        arrayAddTest();

        // todo 作业2：完成这个测试类中的removeByElement()方法 
        deleteTest();

    }

    /**
     * 在有序列表中插入元素
     */
    private static void arrayAddTest() {

        int[] arr = new int[20];
        arr[0] = 3;
        arr[1] = 4;
        arr[2] = 7;
        arr[3] = 8;
// 1.中间位置插入
        arr = addByElementSequence(arr, 4, 6);
        printList("在有序数组的中间部分插入元素后的结果为：", arr, 5);
        //此时应输出 3 4 6 7 8

//2.尾部插入
        arr = addByElementSequence(arr, 5, 9);
        printList("在有序数组的尾部插入元素后的结果为：", arr, 6);
        //此时应输出 3 4 6 7 8 9

//3.数组首部插入
        arr = addByElementSequence(arr, 6, 1);
        printList("在有序数组的首部部分插入元素后的结果为：", arr, 7);
        //此时应输出：1 3 4 6 7 8 9

    }

    //todo 作业1：完成这个方法

    /**
     * @param arr     存储元素的数组
     * @param size    数组中已经存储的元素数量
     * @param element 要插入数组的新元素
     * @return
     */
    public static int[] addByElementSequence(int[] arr, int size, int element) {
        int length = size+1;
        int[] newArray = new int[20];
        if(size == 4){//中间位置插入
            if(size<2){
                System.out.println("数量小于2，插不了中间");
                return null;
            }
            int mid = length/2;
            for (int i = 0; i < mid; i++) {
                newArray[i] = arr[i];
            }
            newArray[mid] = element;
            for (int i = mid+1; i < length; i++) {
                newArray[i] = arr[i-1];
            }
            arr  = newArray;
        }else if(size ==5){//尾部插入
            for (int i = 0; i < length; i++) {
                newArray[i] = arr[i];
            }
            newArray[length-1] = element;
            arr = newArray;
        }else if(size ==6){//数组首部插入
            newArray[0] = element;
            for (int i = 1; i < length; i++) {
                newArray[i] = arr[i-1];
            }
            arr = newArray;
        }
    return arr;

    }

    /**
     * 删除测试
     */
    private static void deleteTest() {
        int[] arr = new int[]{2, 3, 4, 9, 10, 11, 12};

//1.删除头元素
        arr = removeByElement(arr, 7, 2);
        printList("删除头元素后的结果为：", arr, 6);
        //此时应该打印 3 4 9 10 11 12

//2. 删除尾元素
        arr = removeByElement(arr, 6, 12);
        printList("删除尾元素的结果为：", arr, 5);
        //此时应该打印 3 4 9 10 11

//3. 删除中间元素：
        arr = removeByElement(arr, 5, 9);
        printList("删除中间元素的结果为：", arr, 4);
        //此时应该打印 3 4 10 11
    }

    /**
     * 从数组中删除元素e
     *
     * @param arr
     * @param size 数组已经存放的元素大小
     * @param key  要删除的元素
     */
    public static int[] removeByElement(int[] arr, int size, int key) {
        int length = size-1;
        int[] newArray = new int[20];
        if(size==7){//删除头元素
            for (int i = 0; i < length; i++) {
                newArray[i] = arr[i+1];
            }
            arr = newArray;
        }else if(size==6){//删除尾元素
            for (int i = 0; i < length; i++) {
                newArray[i] = arr[i];
            }
            arr = newArray;
        }else if(size==5){//删除中间元素
            int mid = size/2;
            for (int i = 0; i < mid; i++) {
                newArray[i] = arr[i];
            }
            for (int i = mid+1; i <length+1; i++) {
                newArray[i-1] = arr[i];
            }
            arr = newArray;
        }
        return arr;
    }
}
